#Creamos el grafico de densidad de probabilidad

import math
from matplotlib import pyplot
import itertools
import pylab

def importData(name):
    x=[]
    y=[]
    erry=[]
    for line in file(name):
        line=line.replace('\n','')
        val=line.split(' ')
        res=[]
        for v in val:
            if v!='':
                res.append(float(v))
        x.append(res[0])
        y.append(res[1])
        erry.append(res[2])
    return [x,y,erry]
        

#lista de amplitudes a analizar
cuales=['0.05','0.1','0.2','0.5','1','2','5','10','20','40','60','100','200']
cuales=['0.05','1','2','5','10','20','40','60','100','200']
cuales=['0.05','10','40','60','100','200']
cuales=['10']
#Directorio de trabajo
dir='/home/jcfernandez/resultados/eps1/u20/AMPL/W2.0/CONNECT/NEO/connect/'
dir='/Users/jonk/Documents/Trabajo/resultados/eps1/u20/AMPL/W2.0/CONNECT/NEO/'
fig = pyplot.figure()
ax1 = fig.add_subplot(111)
ax1.set_ylabel('Conectividad horizontal')
ax1.set_xlabel(r'$\theta$')
ax1.set_title("Densidad de conectividad")


for j in range(len(cuales)):
    #Nombre del fichero a importar
    name='nconnectA'+cuales[j]+'.dat'
    print 'Fichero: ',name
    [x,y,erry]=importData(dir+name)
#    pyplot.errorbar(x, y, yerr=erry, fmt='ro-')
    ax1.plot(x, y,'o-')


#plot(x,y,'ro')
#pylab.legend(tuple(cuales), shadow = True, loc = (0.5, 0.3))
pylab.legend(tuple(["u100","u20"]), shadow = True, loc = (0.5, 0.3))
pyplot.show()
